#!/usr/bin/env python3
"""
环境配置设置脚本
"""

import os

def create_env_file():
    """创建 .env 文件"""
    env_content = """# 数据库配置
DATABASE_URL=sqlite:///./fortune_telling.db

# JWT 配置
SECRET_KEY=your-secret-key-here-change-this-in-production-2024
ALGORITHM=HS256
ACCESS_TOKEN_EXPIRE_MINUTES=30

# OpenAI 配置
OPENAI_API_KEY=your-openai-api-key-here
OPENAI_BASE_URL=https://api.openai.com/v1
OPENAI_MODEL=gpt-3.5-turbo
OPENAI_ANALYSIS_MODEL=gpt-3.5-turbo

# 应用配置
APP_NAME=AI生辰八字算命系统
DEBUG=True
"""
    
    try:
        with open('.env', 'w', encoding='utf-8') as f:
            f.write(env_content)
        print("✅ .env 文件创建成功！")
        print("请编辑 .env 文件，设置您的 OpenAI API Key")
        return True
    except Exception as e:
        print(f"❌ 创建 .env 文件失败: {e}")
        return False

def check_env_file():
    """检查 .env 文件是否存在"""
    if os.path.exists('.env'):
        print("✅ .env 文件已存在")
        return True
    else:
        print("❌ .env 文件不存在")
        return False

def main():
    print("AI 生辰八字算命系统 - 环境配置")
    print("=" * 40)
    
    if not check_env_file():
        print("\n正在创建 .env 文件...")
        if create_env_file():
            print("\n📝 请按照以下步骤配置：")
            print("1. 打开 .env 文件")
            print("2. 将 'your-openai-api-key-here' 替换为您的实际 OpenAI API Key")
            print("3. 保存文件")
            print("4. 重新运行应用")
        else:
            print("\n❌ 环境配置失败")
    else:
        print("\n✅ 环境配置已完成")

if __name__ == "__main__":
    main()
